/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package com.qp.smsmarketing.listagem;

import com.qp.smsmarketing.entity.Contato;
import com.qp.smsmarketing.entity.GrupoContato;
import java.util.List;
import javax.swing.JOptionPane;
import javax.swing.table.DefaultTableModel;

/**
 *
 * @author joasramos
 */
public class ListagemContato extends DefaultTableModel {

    List<Contato> lista = null;

    public ListagemContato(List<Contato> lista) {
        this.lista = lista;
    }

    @Override
    public int getColumnCount() {
        return 6;
    }

    @Override
    public int getRowCount() {
        if (lista == null) {
            return 0;
        } else {
            return lista.size();
        }
    }

    @Override
    public String getColumnName(int column) {
        String nomeColuna = "";
        if (column == 0) {
            nomeColuna = "Nome";
        } else if (column == 1) {
            nomeColuna = "SobreNome";
        } else if (column == 2) {
            nomeColuna = "Celular";
        } else if (column == 3) {
            nomeColuna = "Aniversario";
        } else if (column == 4) {
            nomeColuna = "Grupo";
        } else if (column == 5) {
            nomeColuna = "Sexo";
        }
        return nomeColuna;
    }

    @Override
    public Object getValueAt(int row, int column) {
        Contato contato = lista.get(row);
        Object objetoContato = null;

        int size = contato.getGrupoContatoList().size();

        if (column == 0) {
            objetoContato = contato.getNomeContato();
        } else if (column == 1) {
            objetoContato = contato.getSobrenomeContato();
        } else if (column == 2) {
            objetoContato = contato.getSmsContato();
        } else if (column == 3) {
            objetoContato = contato.getDataaniContato();
        } else if (column == 4) {
            if (size > 0) {
                objetoContato = contato.getGrupoContatoList().get(0).getGrupoGrupoContato().getNomeGrupo();
            }
        } else if (column == 5) {
            objetoContato = contato.getSexoContato();
        }

        return objetoContato;
    }
}
